#include <iostream>
#include <cctype>
using namespace std;
#include "queen.h"
int main()
{
    char select; // 接受用户的回答
    do
    {
        int n;
        cout << "输入皇后的个数：";
        cin >> n;
        Queen(n);
        while (cin.get() != '\n')
            ; // 跳过当前行后面的字符
        cout << "是否继续(y/n)?";
        cin >> select;            // 用户输入
        select = tolower(select); // 忽略大小写差异
        while (select != 'y' && select != 'n')
        { // 输入错误
            cout << "输入有错，请重新输入(y/n)：";
            cin >> select;
            select = tolower(select);
        }
    } while (select == 'y');
    return 0;
}